package com.wu.rabbitmq.four;

import com.rabbitmq.client.Channel;
import com.rabbitmq.client.ConfirmCallback;
import com.wu.rabbitmq.utils.RabbitMQUtils;

import java.nio.charset.StandardCharsets;
import java.util.UUID;
import java.util.concurrent.ConcurrentNavigableMap;
import java.util.concurrent.ConcurrentSkipListMap;

/**
 * @program: RabbitMQ-Study
 * @author: 吴磊
 * @create: 2021-12-15 10:37
 * 发布确认 模式
 * 使用的时间来比较那种确认方式是最好的
 * 1、单个确认模式
 * 2、批量确认模式
 * 3、异步批量确认
 */
public class ConfirmMessage {
    /**
     * 发布消息的条数
     */
    private final static int MESSAGE_COUNT = 1000;

    public static void main(String[] args) throws Exception {
        /**
         *  1、单个确认
         *  发布1000个单独确认消息耗时：11227 毫秒
         */
        //publishMessageIndividually();
        /**
         * 2、批量确认
         * 发布1000批量确认消息耗时：275 毫秒
         */
        //publishMessageBatch();
        /**
         * 3、异步批量确认
         * 发布1000异步确认消息耗时：33 毫秒
         */
        publishMessageAsync();
    }

    /**
     * 单个确认 : 慢，但是保险，因为知道谁出问题了
     *
     * @throws Exception
     */
    private static void publishMessageIndividually() throws Exception {
        Channel channel = RabbitMQUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();
        long startTime = System.currentTimeMillis();

        for (int i = 0; i < MESSAGE_COUNT; i++) {
            channel.basicPublish("", queueName, null, (i + "").getBytes(StandardCharsets.UTF_8));
            //单个消息就马上进行发布确认
            channel.waitForConfirms();
        }
        long endTime = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "个单独确认消息耗时：" + (endTime - startTime) + " 毫秒");
    }

    /**
     * 批量发布确认 ：快了一点，但是如果一旦出问题了，不知道是哪条消息出问题了
     *
     * @throws Exception
     */
    private static void publishMessageBatch() throws Exception {
        Channel channel = RabbitMQUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();
        long startTime = System.currentTimeMillis();

        //批量确认消息的大小
        int batchSize = 100;

        //批量发送消息，批量发送消息
        for (int i = 1; i <= MESSAGE_COUNT; i++) {
            channel.basicPublish("", queueName, null, (i + "").getBytes(StandardCharsets.UTF_8));
            //达到100条消息的时候确认一次
            if (i % batchSize == 0) {
                channel.waitForConfirms();
            }
        }
        long endTime = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "批量确认消息耗时：" + (endTime - startTime) + " 毫秒");
    }

    /**
     * 异步发布确认
     *
     * @throws Exception
     */
    private static void publishMessageAsync() throws Exception {
        Channel channel = RabbitMQUtils.getChannel();
        //队列的声明
        String queueName = UUID.randomUUID().toString();
        channel.queueDeclare(queueName, false, false, false, null);
        //开启发布确认
        channel.confirmSelect();

        /**
         * 线程安全有序的一个哈希表  适用于高并发的情况下
         * 1、轻松的将序号与消息进行关联
         * 2、轻松的批量删除条目
         * 3、支持高并发（多线程）
         */
        ConcurrentSkipListMap<Long, String> outstandingConfirms = new ConcurrentSkipListMap<>();

        //消息确认成功回调函数
        ConfirmCallback ackCallback = (deliveryTag, multiple) -> {
            if (multiple) {
                //2、删除掉已经确认的消息
                ConcurrentNavigableMap<Long, String> confirmed = outstandingConfirms.headMap(deliveryTag);
                confirmed.clear();
            } else {
                outstandingConfirms.remove(deliveryTag);
            }
            System.out.println("确认的消息：" + deliveryTag);
        };
        //消息确认失败回调函数
        /**
         * deliveryTag  消息的标识
         * multiple  是否是批量确认
         */
        ConfirmCallback nackCallback = (deliveryTag, multiple) -> {
            //3、打印下未确认的消息有哪些
            System.out.println("未确认的消息：" + outstandingConfirms.get(deliveryTag));
        };
        //准备消息的监听器  监听哪些消息成功了，哪些消息失败了
        channel.addConfirmListener(ackCallback, nackCallback);
        long startTime = System.currentTimeMillis();
        //批量发送消息
        for (int i = 0; i < MESSAGE_COUNT; i++) {
            String message = "消息" + i;
            channel.basicPublish("", queueName, null, message.getBytes(StandardCharsets.UTF_8));
            //1、记录下所有要发送的消息  消息的总和
            outstandingConfirms.put(channel.getNextPublishSeqNo(), message);
        }
        long endTime = System.currentTimeMillis();
        System.out.println("发布" + MESSAGE_COUNT + "异步确认消息耗时：" + (endTime - startTime) + " 毫秒");
    }
}
